import os
from dotenv import load_dotenv

# 加载.env文件中的环境变量
load_dotenv()

CONFIG = {
    # 模型配置
    "MODELS": {
        "GLM_4_FLASH": {
            "name": os.getenv("GLM_4_FLASH"),
            "api_url": None,  # 如果需要URL，可填入实际值
            "token": os.getenv("GLM_4_FLASH_TOKEN"),
        },
        "GLM_4_PLUS": {
            "name": os.getenv("GLM_4_PLUS"),
            "api_url": None,
            "token": os.getenv("GLM_4_PLUS_TOKEN"),
        },
        "GLM_4_LONG": {
            "name": os.getenv("GLM_4_LONG"),
            "api_url": None,
            "token": os.getenv("GLM_4_LONG_TOKEN"),
        },
        "GLM_ZERO_PREVIEW": {
            "name": os.getenv("GLM_ZERO_PREVIEW"),
            "api_url": None,
            "token": os.getenv("GLM_ZERO_PREVIEW_TOKEN"),
        },
    },
}

# GLM-4模型列表
GLM_4_MODEL_LIST = ["glm-4-plus", "glm-4-0520", "glm-4", "glm-4-air", "glm-4-airx", "glm-4-long", "glm-4-flashx", "glm-4-flash", "glm-zero-preview"]
GLM_4_MODEL_UPPER_LIST = ["GLM_4_PLUS", "GLM_4_0520", "GLM_4", "GLM_4_AIR", "GLM_4_AIRX", "GLM_4_LONG", "GLM_4_FLASHX", "GLM_4_FLASH", "GLM_ZERO_PREVIEW"]